package com.javaee.wordtree.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.javaee.wordtree.entity.Chat;
import com.javaee.wordtree.mapper.ChatMapper;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.Date;
import java.util.List;

/**
 * 聊天服务类
 */
@Service
public class ChatService extends ServiceImpl<ChatMapper, Chat> {
    
    @Resource
    private ChatMapper chatMapper;
    
    /**
     * 获取用户的所有聊天记录
     * @param userId 用户ID
     * @return 聊天列表
     */
    public List<Chat> getUserChats(String userId) {
        return chatMapper.findChatsByUserId(userId);
    }
    
    /**
     * 创建新聊天
     * @param userId 用户ID
     * @param title 聊天标题
     * @return 新创建的聊天对象
     */
    public Chat createChat(String userId, String title) {
        Chat chat = new Chat();
        chat.setUserId(userId);
        chat.setTitle(title);
        Date now = new Date();
        chat.setCreatedAt(now);
        chat.setUpdatedAt(now);
        
        save(chat); // 使用MyBatis-Plus提供的save方法保存对象
        return chat;
    }
    
    /**
     * 更新聊天标题
     * @param chatId 聊天ID
     * @param title 新标题
     * @return 是否更新成功
     */
    public boolean updateChatTitle(Integer chatId, String title) {
        return chatMapper.updateChatTitle(chatId, title) > 0;
    }
    
    /**
     * 更新聊天的更新时间
     * @param chatId 聊天ID
     * @return 是否更新成功
     */
    public boolean updateChatTime(Integer chatId) {
        return chatMapper.updateChatTime(chatId) > 0;
    }
    
    /**
     * 根据ID获取聊天记录
     * @param chatId 聊天ID
     * @return 聊天对象
     */
    public Chat getChatById(Integer chatId) {
        return getById(chatId); // 使用MyBatis-Plus提供的getById方法
    }
    
    /**
     * 删除聊天记录
     * @param chatId 聊天ID
     * @return 是否删除成功
     */
    public boolean deleteChat(Integer chatId) {
        return removeById(chatId); // 使用MyBatis-Plus提供的removeById方法
    }
    
    /**
     * 检查聊天是否属于指定用户
     * @param chatId 聊天ID
     * @param userId 用户ID
     * @return 是否属于该用户
     */
    public boolean isChatBelongToUser(Integer chatId, String userId) {
        LambdaQueryWrapper<Chat> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Chat::getChatId, chatId).eq(Chat::getUserId, userId);
        return count(queryWrapper) > 0;
    }
}
